#include <iostream>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string>
#include <functional>
#include "Mutex.hpp"
#include "Log.hpp"
#include "InetAddr.hpp"

using namespace LogModule;
using namespace MutexModule;

const int defaultfd = -1;

using func_t =std::function<std::string(const std::string&,InetAddr&)>;

class UdpServer
{
public:
    UdpServer(uint16_t port,func_t func)
        : _sockfd(defaultfd), _port(port), _isrunning(false),_func(func)
    {
    }

    void Init()
    {
        // 1.创建套接字
        _sockfd = socket(AF_INET, SOCK_DGRAM, 0);
        if (_sockfd < 0)
        {
            LOG(LogLevel::Fatal) << "socket error";
            exit(1);
        }
        LOG(LogLevel::Info) << "socket created success";

        // 2.绑定socket信息，包括IP地址和端口号
        // 2.1 填充socketaddr_in结构体
        struct sockaddr_in local;
        local.sin_family = AF_INET;
        // 我会不会把ip和port发送给对方？会
        // 所以，IP信息和端口号一定要发送到网络！
        // 本地格式->网络序列
        local.sin_port = htons(_port);
        // Ip也是如此，//1.把ip转成uint32_t 2.把uint32_t转成网络序列 ->in_addr_t inet_addr(const char* cp)
        //local.sin_addr.s_addr = inet_addr(_ip.c_str());

        //由于公网Ip没有配置到我的IP上，所以不能被bind
        //bind 127.0.0.1 ||内网IP成功，
        //但server端bind 127.0.0.1，client端用内网IP访问，（或者两者颠倒）就不成功
        //就说明如果server端显式的bind Ip地址，那么client端必须使用server端bind的地址才能访问成功
        //server端不建议bind特定的IP地址，而是使用INADDR_ANY，这样可以接收来自任何IP的请求
        local.sin_addr.s_addr=INADDR_ANY;

        // 那么为什么服务器端要显式的bind呢？IP和端口必须是众所周知且不能轻易改变的！
        int n = bind(_sockfd, (struct sockaddr *)&local, sizeof(local));
        if (n < 0)
        {
            LOG(LogLevel::Fatal) << "bind error";
            exit(2);
        }

        LOG(LogLevel::Info) << "bind success";
    }

    void Start()
    {
        _isrunning=true;
        while(_isrunning)
        {
            //1.收消息,client为什么要个服务器发送消息啊？不就是让服务端处理数据。
            char buffer[1024];
            struct sockaddr_in peer;
            socklen_t len=sizeof(peer);
            ssize_t s=recvfrom(_sockfd,buffer,sizeof(buffer)-1,0,(struct sockaddr*)&peer,&len);
            if(s>0)
            {
                // uint16_t peer_port=ntohs(peer.sin_port);// 从网络中拿到的！网络序列
                // std::string peer_ip=inet_ntoa(peer.sin_addr);//4字节网络风格的IP -> 点分十进制的字符串风格的IP

                InetAddr addr(peer);

                buffer[s]=0;

                std::string result=_func(buffer,addr);

                //知道是谁发的消息
                //std::cout<<"["<<peer_ip<<":"<<peer_port<<"]"<<buffer<<std::endl;

                // //2.发消息
                // std::string echo_string="server echo@: ";
                // echo_string+=buffer;
                sendto(_sockfd,result.c_str(),result.size(),0,(struct sockaddr*)&peer,len);
            }
        }
    }

    ~UdpServer()
    {
    }

private:
    int _sockfd;
    uint16_t _port;
    //std::string _ip; // 用的是字符串风格的IP地址，点分十进制
    bool _isrunning;
    func_t _func;// 服务器的回调函数，用来进行对数据进行处理
};
